package com.demo.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.demo.common.CustomException;
import com.demo.common.R;
import com.demo.dao.CategoryDao;
import com.demo.dao.DishDao;
import com.demo.dao.SetmealDao;
import com.demo.dao.SetmealDishDao;
import com.demo.domain.Category;
import com.demo.domain.Setmeal;
import com.demo.domain.SetmealDish;
import com.demo.dto.SetmealDto;
import com.demo.service.SetmealDishService;
import com.demo.service.SetmealService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

@Service
@Slf4j
public class SetmealServiceImpl extends ServiceImpl<SetmealDao, Setmeal> implements SetmealService {

    @Autowired
    private SetmealDishService setmealDishService;

    @Autowired
    private CategoryDao categoryDao;

    @Autowired
    private SetmealDao setmealDao;

    /**
     * 分页条件查询所有数据
     *
     * @param page
     * @param pageSize
     * @param name
     * @return
     */
    @Override
    public R<Page> page(int page, int pageSize, String name) {
        //创建分页构造器
        Page<Setmeal> pageInfo = new Page<>(page, pageSize);
        Page<SetmealDto> setmealDtoPage = new Page<>(page, pageSize);

        //创建查询构造器
        LambdaQueryWrapper<Setmeal> lqw = new LambdaQueryWrapper<>();
        //添加模糊查询
        lqw.like(!StringUtils.isEmpty(name), Setmeal::getName, name);
        //添加排序规则
        lqw.orderByDesc(Setmeal::getUpdateTime);
        //执行查询
        this.page(pageInfo, lqw);

        //将pageInfo中的基本数据复制到setmealDtoPage中
        BeanUtils.copyProperties(pageInfo, setmealDtoPage, "records");
        //获取到pageInfo中records的数据
        List<Setmeal> records = pageInfo.getRecords();
        //对records里的数据进行修改
        List<SetmealDto> List = records.stream().map((item) -> {
            SetmealDto setmealDto = new SetmealDto();
            BeanUtils.copyProperties(item, setmealDto);
            //获取item里套餐分类的id
            Long categoryId = item.getCategoryId();
            //通过categoryId创建category对象
            Category category = categoryDao.selectById(categoryId);
            if (category != null) {
                //重新对item的categoryName赋值
                setmealDto.setCategoryName(category.getName());
            }

            return setmealDto;
        }).collect(Collectors.toList());

        setmealDtoPage.setRecords(List);

        return R.success(setmealDtoPage);
    }

    /**
     * 分页查询    代码优化
     * 多表条件分页查询
     *
     * @param setmealDtoPage
     * @param setmealDto
     */
    @Override
    public void pageByParams(Page<SetmealDto> setmealDtoPage, SetmealDto setmealDto) {
        setmealDao.pageByParams(setmealDtoPage, setmealDto);
    }


    /**
     * 添加套餐数据
     *
     * @param setmealDto
     * @return
     */
    @Override
    @Transactional
    public R<String> setmealSave(SetmealDto setmealDto) {
        //保存基本信息
        this.save(setmealDto);

        //获取setmealDto中的setmealDishes集合信息
        List<SetmealDish> setmealDishes = setmealDto.getSetmealDishes();
        //因为集合信息中缺少套餐setmealId，需要补全信息
        setmealDishes.stream().map((item) -> {
            item.setSetmealId(setmealDto.getId());
            return item;
        }).collect(Collectors.toList());

        //将集合setmealDishes数据添加到SetmealDish表中
        setmealDishService.saveBatch(setmealDishes);

        return R.success("保存信息成功");
    }


    /**
     * 单删和批量删除套餐信息
     *
     * @param ids
     * @return
     */
    @Override
    @Transactional
    public R<String> deleteSetmeal(Long[] ids) {
        LambdaUpdateWrapper<Setmeal> setmeallqw = new LambdaUpdateWrapper<>();
        //查询要删除的套餐
        setmeallqw.in(Setmeal::getId, Arrays.asList(ids));
        //查询当前套餐的状态是否为起售
        setmeallqw.eq(Setmeal::getStatus, 1);
        int count = this.count(setmeallqw);
        //当套餐为起售是，抛出一个异常提示
        if (count > 0) {
            throw new CustomException("当前套餐正在售卖中，无法删除");
        } else {
            //不为起售时即可删除套餐信息
            this.removeByIds(Arrays.asList(ids));
            //还需要删除菜品关系表中的菜品信息
            LambdaQueryWrapper<SetmealDish> lqw = new LambdaQueryWrapper<>();
            lqw.in(SetmealDish::getSetmealId, Arrays.asList(ids));
            setmealDishService.remove(lqw);
            return R.success("删除成功");
        }

    }

    /**
     * 修改套餐状态
     *
     * @param status
     * @param ids
     * @return
     */
    @Override
    public R<String> setmealStatusByStatus(@PathVariable int status, Long[] ids) {
        for (Long id : ids) {
            LambdaUpdateWrapper<Setmeal> lqw = new LambdaUpdateWrapper<>();
            lqw.eq(Setmeal::getId, id).set(Setmeal::getStatus, status);
            this.update(lqw);
        }
        return R.success("修改成功");
    }

    /**
     * 通过id回显数据
     *
     * @param id
     * @return
     */
    @Override
    public R<SetmealDto> querySetmealById(@PathVariable Long id) {
        //查询与id匹配的套餐基本数据
        Setmeal setmeal = this.getById(id);
        log.info(setmeal.toString());
        //因为缺少套餐名称与菜品信息，所以需要创建Dto来添加信息
        SetmealDto setmealDto = new SetmealDto();
        //将基本信息复制到Dto中
        BeanUtils.copyProperties(setmeal, setmealDto);
        //获取基本信息中的套餐类型id
        Long categoryId = setmeal.getCategoryId();
        //通过套餐类型id获取到category对象
        Category category = categoryDao.selectById(categoryId);
        //通过category对象获取到套餐名称并赋值到setmealDto的categoryName属性上
        setmealDto.setCategoryName(category.getName());

        //通过套餐菜品关系表，传输套餐id来获取菜品详情信息
        LambdaQueryWrapper<SetmealDish> lqw = new LambdaQueryWrapper<>();
        lqw.eq(SetmealDish::getSetmealId, setmealDto.getId());
        List<SetmealDish> list = setmealDishService.list(lqw);

        //将获取到的菜品详情信息赋值到setmealDto的属性中
        setmealDto.setSetmealDishes(list);

        //最后返回setmealDto
        return R.success(setmealDto);
    }

    /**
     * 优化代码
     * 一对多查询信息
     * 通过id回显数据
     *
     * @param id
     * @return
     */
    @Override
    public SetmealDto queryById(Long id) {
        return setmealDao.queryById(id);
    }


    /**
     * 修改套餐信息
     *
     * @param setmealDto
     * @return
     */
    @Override
    @Transactional
    public R<String> editSetmeal(@RequestBody SetmealDto setmealDto) {
        //保存基本信息
        this.updateById(setmealDto);
        //获取到setmealDto中的套餐id，来修改分类表中的套餐名称
        Long categoryId = setmealDto.getCategoryId();
        Category category = categoryDao.selectById(categoryId);
        setmealDto.setCategoryName(category.getName());

        //删除原本套餐中的菜品信息
        LambdaQueryWrapper<SetmealDish> lqw = new LambdaQueryWrapper<>();
        lqw.eq(SetmealDish::getSetmealId, setmealDto.getId());
        setmealDishService.remove(lqw);

        //再添加修改后的菜品信息
        //获取到setmealDto中的setmealDishes数据
        List<SetmealDish> setmealDishes = setmealDto.getSetmealDishes();
        setmealDishes = setmealDishes.stream().map((item) -> {
            item.setSetmealId(setmealDto.getId());
            return item;
        }).collect(Collectors.toList());

        setmealDishService.saveBatch(setmealDishes);
        return R.success("修改成功");
    }


    /**
     * App端展示套餐信息
     * @param setmeal
     * @return
     */
    @Override
    public R<List<Setmeal>> setmealListApi(Setmeal setmeal) {
        LambdaQueryWrapper<Setmeal> lqw = new LambdaQueryWrapper<>();
        lqw.eq(setmeal.getCategoryId() != null,Setmeal::getCategoryId, setmeal.getCategoryId());
        lqw.eq(setmeal.getStatus() != null,Setmeal::getStatus, setmeal.getStatus());
        lqw.orderByDesc(Setmeal::getUpdateTime);

        //获取套餐集合数据，不过缺少菜品信息
        List<Setmeal> list = this.list(lqw);

        return R.success(list);

    }
}
